import requests


class IHRMEmpManageApi:

    # 添加员工
    @classmethod
    def add_emp_api(cls, token, add_emp_body):
        url = "http://ihrm2-test.itheima.net/api/sys/user"
        header = {"Content-Type": "application/json", "Authorization": token}
        resp = requests.post(url=url, headers=header, json=add_emp_body)
        return resp

    # 查询员工
    @classmethod
    def query_emp_api(cls, emp_id, token):
        url = "http://ihrm2-test.itheima.net/api/sys/user/" + emp_id
        header = {"Authorization": token}
        resp = requests.get(url=url, headers=header)
        return resp

    # 修改员工
    @classmethod
    def modify_emp_api(cls, emp_id, token, modify_emp_body):
        url = "http://ihrm2-test.itheima.net/api/sys/user/" + emp_id
        header = {"Content-Type": "application/json",
                  "Authorization": token}

        resp = requests.put(url=url, headers=header, json=modify_emp_body)
        return resp

    # 删除员工
    @classmethod
    def del_emp_api(cls, emp_id, token):
        url = "http://ihrm2-test.itheima.net/api/sys/user/" + emp_id
        header = {"Authorization": token}
        resp = requests.delete(url=url, headers=header)
        return resp


